flatland space stations hackerrank solution in python